from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension

setup(
        name = 'fps',
        ext_modules = [
            CUDAExtension('fps', [
                'src/pointnet2_api.cpp',
                'src/ball_query.cpp',
                'src/ball_query_gpu.cu',
                'src/group_points.cpp',
                'src/group_points_gpu.cu',
                'src/interpolate.cpp',
                'src/interpolate_gpu.cu',
                'src/sampling.cpp',
                'src/sampling_gpu.cu',
            ],
                          extra_compile_args = { 'cxx' : ['-g'],
                                                 'nvcc': ['-O2'] })
        ],
        cmdclass = { 'build_ext': BuildExtension }
)
